﻿//
// (C) Copyright 2009 Irantha Suwandarathna (iranthas@hotmail.com)
// All rights reserved.
//

/* Copyright (c) 2001-2008, The HSQL Development Group
 * All rights reserved.
 *
 * Redistribution and use _in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice, this
 * list of conditions and the following disclaimer.
 *
 * Redistributions _in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer _in the documentation
 * and/or other materials provided with the distribution.
 *
 * Neither the name of the HSQL Development Group nor the names of its
 * contributors may be used to endorse or promote products derived from this
 * software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL HSQL DEVELOPMENT GROUP, HSQLDB.ORG,
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
#if !SILVERLIGHT
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.IO;
using System.IO.Compression;
using EffiProz.Core.Lib;
using EffiProz.Core.Persist;
using EffiProz.Core.RowIO;
using EffiProz.Core.Errors;

namespace EffiProz.Core.ScriptIO
{
    public class ScriptWriterEncode : ScriptWriterText
    {
        Crypto crypto;
        MemoryStream memOut;
        BinaryWriter byteOut;
        Stream cStream;

        public ScriptWriterEncode(Database db, String file, bool includeCached,
                                  Crypto crypto)
            : base(db, file, includeCached, true, false)
        {

            try
            {
                cStream = fileStreamOut = crypto.getOutputStream(fileStreamOut);
                fileStreamOut = new GZipStream(fileStreamOut, CompressionMode.Compress);
            }
            catch (IOException e)
            {
                throw Error.error(e, ErrorCode.FILE_IO_ERROR,
                                  ErrorCode.M_Message_Pair, new Object[] {
                e.Message, outFile
            });
            }
        }

        public ScriptWriterEncode(Database db, String file, Crypto crypto)
            : base(db, file, false, true, false)
        {

            this.crypto = crypto;
            memOut = new MemoryStream();
            byteOut = new BinaryWriter(memOut);
        }

        protected override void finishStream()
        {         
            fileStreamOut.Flush();
        }

        public override void  writeRowOutToFile()
        {

            lock (fileStreamOut)
            {
                if (byteOut == null)
                {
                    base.writeRowOutToFile();

                    return;
                }

                byte[] encrypted = crypto.encode(rowOut.getBuffer(), 0, rowOut.size());
                int count = encrypted.Length;
              
                byteOut.Seek(0, SeekOrigin.Begin);
                byteOut.Write((int)count);
                byteOut.Write(encrypted, 0, count);
                byteOut.Flush();

                fileStreamOut.Write(memOut.GetBuffer(), 0, count + 4);
            }
        }
    }
}

#endif
